# pip install zhipuai 请先在终端进行安装

from zhipuai import ZhipuAI

from 测试项目.示例代码 import response


class 智普ai初始化:
    def __init__(self ,智普api_key, 模型名称, 最大tokens数量):
        self.智普ai = ZhipuAI(api_key=智普api_key)
        self.模型名称=模型名称
        self.最大tokens数量=最大tokens数量
        self.对话对象名 = "system"
        self.对话提示词="你是一个乐于解答各种问题的助手，你的任务是为用户提供专业、准确、有见地的建议。"
        self.对话用户名="user"


    def 询问(self,对话内容,对话对象名=None,对话提示词=None,对话用户名=None,):
        if 对话对象名:
            self.对话对象名=对话对象名
        if 对话提示词:
            self.对话提示词 =对话提示词
        if 对话用户名:
            self.对话用户名 =对话用户名
        if not 对话内容:
            response("对话内容不能为空")

        回答 = self.智普ai.chat.completions.create(
            model=self.模型名称,
            messages=[
                {
                    "role": self.对话对象名,
                    "content": self.对话提示词
                },
                {
                    "role": self.对话用户名,
                    "content": 对话内容
                }
            ],
            top_p=0.7,
            temperature=0.95,
            max_tokens=self.最大tokens数量,
            tools=[{"type": "web_search", "web_search": {"search_result": True}}],
            stream=True
        )
        return 回答

#
# for trunk in response:
#     print(trunk)

